multicontext imitation learning